package com.sh.data.engine.application.model.dto.dataasset;

import com.sh.data.engine.domain.shims.db.model.TableInfoDomain;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.apache.commons.lang3.StringUtils;

import java.util.StringJoiner;

/**
 * @author: mengzheng.mhc
 * @date: 2024/6/26 10:58
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
//@Builder(toBuilder = true)
public class TableInfoDTO {

    /**
     * 库名
     */
    private String dbName;

    /**
     * schema
     */
    private String schema;

    /**
     * 表id，hive库有
     */
    private Long tblId;

    /**
     * 表名
     */
    private String tableName;

    /**
     * 表注释
     */
    private String tableComment;

    /**
     * 表类型
     */
    private TblType tblType;

    /**
     * 最新更新时间
     */
    private Long lastDdlTime;

    /**
     * 是否是分区表
     */
    private boolean isPartitionTable;

    /**
     * 密级
     */
    private Long dataSecretId;


    @Override
    public String toString() {
        return new StringJoiner(", ", TableInfoDomain.class.getSimpleName() + "[", "]")
            .add("dbName='" + dbName + "'")
            .add("schema='" + schema + "'")
            .add("tblId=" + tblId)
            .add("tableName='" + tableName + "'")
            .add("tableComment='" + tableComment + "'")
            .add("tblType=" + tblType)
            .add("lastDdlTime=" + lastDdlTime)
            .add("isPartitionTable=" + isPartitionTable)
            .toString();
    }

    public enum TblType {
        /**
         * 内表
         */
        MANAGED_TABLE,
        /**
         * 外表
         */
        EXTERNAL_TABLE,
        /**
         * 视图
         */
        VIEW;

        public static TblType from(String tblType) {
            for (TblType value : TblType.values()) {
                if (StringUtils.equalsIgnoreCase(tblType, value.name())) {
                    return value;
                }
            }
            return null;
        }
    }
}
